from configparser import ConfigParser, NoSectionError, NoOptionError
from dotenv import load_dotenv

import os

# 如果存在环境变量的文件，则加载配置到环境变量
if os.path.exists("settings.env"):
    load_dotenv("settings.env")

os_env = os.environ

def read_config(filename: str) -> ConfigParser:
    """
    从文件中读取配置信息

    Parameters
    ----------
    filename : str, 配置文件
    """
    # 实例化对象
    config = ConfigParser()
    env = os.environ.get('ENV', 'dev').lower()
    if env == 'dev':
        filename = "dev-config.ini"
    elif env == 'prod':
        filename = "prod-config.ini"
    elif env == 'test1':
        filename = "test.ini"
    elif env == 'test2':
        filename = "test-y.ini"
    if not os.path.exists(filename):
        raise FileNotFoundError(f"配置文件 {filename} 不存在")
    config.read(filename, encoding="utf-8")
    return config


def get_config(config: ConfigParser, section: str, key: str):
    """
    根据指定section和key获取value

    Parameters
    ----------
    config:  ConfigParser(), 配置实例对象
    section: str, 配置文件中的区域
    key:     str, 配置的参数名
    """
    # 优先从环境变量中获取配置参数, 没有的话再从配置文件中获取
    value = os_env.get(key, "")
    if not value:
        try:
            value = config.get(section, key)
        except (NoOptionError, NoSectionError):
            # 没有的话就返回None
            value = None
    return value

def config(section: str, key: str):
    config_file = read_config("")
    return get_config(config_file, section, key)

if __name__ == '__main__':
    print(config( "edge_driver_path", "path"))
